/*
 * Copyright (C) 2009 Leandro Aparecido <lehphyro@gmail.com>
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.metalframework.serialization;

import java.lang.annotation.*;
import java.util.*;

import javax.persistence.*;

import org.metalframework.annotation.*;
import org.metalframework.util.*;
import org.springframework.aop.framework.*;
import org.springframework.aop.support.*;
import org.springframework.beans.*;
import org.springframework.beans.factory.config.*;
import org.springframework.core.*;

@InfrastructureComponent
public class ExposeProxyBeanPostProcessor implements BeanPostProcessor, Ordered {

	private List<Class<? extends Annotation>> markerAnnotations;
	
	public ExposeProxyBeanPostProcessor() {
		markerAnnotations = new ArrayList<Class<? extends Annotation>>();
		markerAnnotations.add(ApplicationComponent.class);
		markerAnnotations.add(Entity.class);
	}

	public Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException {
		// Do nothing
		return bean;
	}

	public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
		if (bean instanceof Advised && isEligible(bean)) {
			Advised advised = (Advised) bean;
			advised.setExposeProxy(true);
		}
		
		return bean;
	}
	
	public int getOrder() {
		return Ordered.LOWEST_PRECEDENCE;
	}
	
	protected boolean isEligible(Object bean) {
		for (Class<? extends Annotation> item : markerAnnotations) {
			if (AnnotationUtils.getAnnotation(AopUtils.getTargetClass(bean), item) != null) {
				return true;
			}
		}
		return false;
	}
}
